package main

import (
	"dir/server"
	"fmt"

	"github.com/gin-gonic/gin"
)

func main() {
	if server.V站点属性 == server.V源站 {
		gin.SetMode(gin.ReleaseMode)
		r := gin.Default()
		r.Use(server.Ip白名单())
		r.GET("/dirs", func(c *gin.Context) {
			dirs := server.F路径列表()
			c.JSON(200, dirs)
		})
		r.GET("/file", func(c *gin.Context) {
			path := c.Query("path")
			//此处限制只能是以下目录里面的，并且限制ip必须为本地
			l := len([]rune(server.V目录))
			if path[:l] != server.V目录[:l] {
				c.JSON(200, "目录异常")
				fmt.Println("目录异常")
				return
			} else {
				c.File(path)
			}
		})
		if err := r.Run(":9988"); err != nil {
			fmt.Printf("startup service failed, err:%+v\n", err)
		}
	} else {

		//server.F路径列表()
		server.F开启任务()

	}

}
